from haystack import indexes

from goods.models import SKU


class SKUIndex(indexes.SearchIndex, indexes.Indexable):
    """创建sku数据索引模型类"""
    # 设置复合字段(数据库模型类的多个字段可以使用这一个index字段来进行查询),使用use_template时,
    # 会在项目的templates中去寻找相关的文件 设置三级目录 search/indexes/goods/sku(模型类)_text(索引字段).txt
    text = indexes.CharField(document=True, use_template=True)
    # 单字段所以 使用model_attr表示索引的字段在数据库表中的对应字段
    id = indexes.IntegerField(model_attr='id')
    name = indexes.CharField(model_attr='name')
    price = indexes.DecimalField(model_attr='price')
    default_image_url = indexes.CharField(model_attr='default_image_url')
    comments = indexes.IntegerField(model_attr='comments')

    def get_model(self):
        """返回建立索引的模型类"""
        return SKU

    def index_queryset(self, using=None):
        """返回要建立索引的数据查询集"""
        return self.get_model().objects.filter(is_launched=True)
